home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / lib / ruby / 1.8 / find.rb < prev    next >
Text File  |  2007-02-12  |  2KB  |  80 lines
  1. #
  2. # find.rb: the Find module for processing all files under a given directory.
  3. #
  4.  
  5. #
  6. # The +Find+ module supports the top-down traversal of a set of file paths.
  7. #
  8. # For example, to total the size of all files under your home directory,
  9. # ignoring anything in a "dot" directory (e.g. $HOME/.ssh):
  10. #
  11. #   require 'find'
  12. #
  13. #   total_size = 0
  14. #
  15. #   Find.find(ENV["HOME"]) do |path|
  16. #     if FileTest.directory?(path)
  17. #       if File.basename(path)[0] == ?.
  18. #         Find.prune       # Don't look any further into this directory.
  19. #       else
  20. #         next
  21. #       end
  22. #     else
  23. #       total_size += FileTest.size(path)
  24. #     end
  25. #   end
  26. #
  27. module Find
  28.  
  29.   #
  30.   # Calls the associated block with the name of every file and directory listed
  31.   # as arguments, then recursively on their subdirectories, and so on.
  32.   #
  33.   # See the +Find+ module documentation for an example.
  34.   #
  35.   def find(*paths) # :yield: path
  36.     paths.collect!{|d| d.dup}
  37.     while file = paths.shift
  38.       catch(:prune) do
  39.     yield file.dup.taint
  40.         next unless File.exist? file
  41.     begin
  42.       if File.lstat(file).directory? then
  43.         d = Dir.open(file)
  44.         begin
  45.           for f in d
  46.         next if f == "." or f == ".."
  47.         if File::ALT_SEPARATOR and file =~ /^(?:[\/\\]|[A-Za-z]:[\/\\]?)$/ then
  48.           f = file + f
  49.         elsif file == "/" then
  50.           f = "/" + f
  51.         else
  52.           f = File.join(file, f)
  53.         end
  54.         paths.unshift f.untaint
  55.           end
  56.         ensure
  57.           d.close
  58.         end
  59.       end
  60.         rescue Errno::ENOENT, Errno::EACCES
  61.     end
  62.       end
  63.     end
  64.   end
  65.  
  66.   #
  67.   # Skips the current file or directory, restarting the loop with the next
  68.   # entry. If the current file is a directory, that directory will not be
  69.   # recursively entered. Meaningful only within the block associated with
  70.   # Find::find.
  71.   #
  72.   # See the +Find+ module documentation for an example.
  73.   #
  74.   def prune
  75.     throw :prune
  76.   end
  77.  
  78.   module_function :find, :prune
  79. end
  80.